#include "mystdio.h"
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <assert.h>
#include <unistd.h>
#include <stdlib.h>

//fopen("/a/b/c.txt","w");
//fopen("/a/b/c.txt","a");
//fopen("/a/b/c.txt","r");  这里实现3中方式
MY_FILE* my_fopen(const char* path, const char* mode)
{
    //1.识别标志位
    int flag=0;
    if(strcmp(mode,"r") == 0) flag |= O_RDONLY; 
    else if(strcmp(mode,"w") == 0) flag |= (O_CREAT | O_WRONLY | O_TRUNC);
    else if(strcmp(mode,"a") == 0) flag |= (O_CREAT | O_WRONLY | O_APPEND);
    else 
    {
        //其它操作
        //如"r+","w+","a+"
    }
    //2.尝试打开文件
    mode_t m = 0666;
    int fd=0;

    //open函数有两种接口，一种是打开已有文件，另一种是创建并打开文件
    if(flag & O_CREAT) fd = open(path,flag,m);
    else fd = open(path,flag);

    //打开失败返回NULL
    if(fd<0) return NULL;

    //3.我们要给用户返回MY_FILE结构体对象，所以我们要先进行构建
    MY_FILE* mf = (MY_FILE*)malloc(sizeof(MY_FILE));
    if(mf == NULL)
    {
        close(fd);
        return NULL; 
    }

    //4.初始化MY_FILE对象
    mf->fd = fd;
    mf->flags = 0;
    mf->flags |= BUFF_LINE;//刷新方式采用行缓冲

    //mf->outputbuffer[0]=0;//初始化缓冲区
    memset(mf->outputbuffer,'\0',sizeof(mf->outputbuffer));

    //5.返回打开的文件
    return mf;
}

int my_fflush(MY_FILE* fp)
{
    assert(fp);
    //使用系统调用接口，将缓冲区的数据刷新到文件描述符为fd的文件中
    //只有调用了my_fflush函数，才会将缓冲区的数据刷新到指定文件中
    write(fp->fd, fp->outputbuffer, fp->current);
    //TODO
    fp->current = 0;

    fsync(fp->fd); //将数据同步到外部存储设备
    return 0;
}

//fwrite函数应该怎么实现呢？
//const void* ptr:要写到打开文件的数据的首地址，这里是缓冲区数据首地址，
//因为我们要把缓冲区的数据写入到所打开的文件中
//size_t size:要写出数据的基本单元的字节大小，如你向文件写入的数据是整型，那么
//size就等于sizeof(int)
//size_t nmemb:要写出数据的基本单元个数，即你要写入几个int类型的数据
//FILE* stream:打开的文件指针
//返回值返回的是实际写出到文件的基本单元的个数
//使用方法：size_t count = fwrite(buffer,sizeof(char),5,fp);
size_t my_fwrite(const void* ptr, size_t size, size_t nmemb, MY_FILE* stream)
{
    //1.缓存区如果已经满了，就直接写入文件
    if(stream->current == NUM) my_fflush(stream);

    //2.根据缓冲区剩余情况，进行数据拷贝
    //【先将用户数据写入缓冲区】
    size_t user_size = size * nmemb;//用户写入数据的总字节大小
    size_t my_size = NUM - stream->current;//缓冲区还可以写入的数据大小

    size_t written = 0;
    if(my_size >= user_size)//缓冲区可以写入的大小大于用户需要写入的大小，可以写入
    {
        //把数据写入缓冲区
        memcpy(stream->outputbuffer+stream->current, ptr, user_size);
        //3.更新计数器字段
        stream->current += user_size;
        written = user_size;
    }
    else 
    {
        //当缓冲区的大小小于用户写入数据的大小时，我们只能写入
        //缓冲区剩余大小的数据
        memcpy(stream->outputbuffer+stream->current, ptr, my_size);
        //3.更新计数器字段
        stream->current += my_size;
        written = my_size;
    }

    //4.开始计划刷新
    //不发生刷新的本质，就是只把数据写入到了缓冲区中，但是还没有写入指定文件，就是不进行IO操作
    //所以my_fwrite函数调用会非常快，数据会暂时存储到缓冲区中
    if(stream->flags & BUFF_ALL)
    {
        if(stream->current == NUM) my_fflush(stream);
    }
    else if(stream->flags & BUFF_ALL)
    {
        if(stream->outputbuffer[stream->current-1] == '\n') my_fflush(stream);
    }
    else 
    {
        //TODO
    }

    return written;
}


int my_fclose(MY_FILE* fp)
{
    //当我们关闭文件的时候，会帮我们刷新缓冲区
    assert(fp);
    //1.冲刷缓冲区
    if(fp->current > 0) my_fflush(fp);
    //2.关闭文件
    close(fp->fd);
    //3.释放堆空间
    free(fp);
    //指针置NULL
    fp = NULL;

    return 0;
}
